#coding=utf-8
'''
# @Name:log
# @author:mysouil
# @Description:日志
# @DATE:2022/9/22 1:19
# @Modified By:
'''
from flask import request
import logging
import os
import logging.handlers

class LoggingFile(logging.Formatter):
    '''
    自定义的日志输出格式
    '''
    def format(self, record):
        # 在日志中记录请求地址
        record.url = request.url
        # 在日志中记录客户端的地址
        record.remote_addr = request.remote_addr
        
        return super().format(record)

# 创建一个logger对象
def create_logger(app):
    '''
    设置日志配置
    :param app:Flask中的app对象
    :return:
    '''
    
    logging_file = app.config['LOGGING_FILE_DIR']  # 日志文件所在目录
    max_logging_size = app.config['LOGGING_FILE_MAX_BYTES'] # 日志文件最大的大小
    logging_file_backup = app.config['LOGGING_FILE_BACKUP'] # 保留备份的日志文件个数
    LOGGING_LEVEL =  app.config['LOGGING_LEVEL'] # 默认的日志级别
    
    # 设置日志的输出格式(针对文件的输出)
    logging_format = LoggingFile('[%(asctime)s]  %(remote_addr)s address[%(url)s] \t %(levelname)s %(module)s %(lineno)d : %(message)s')
    
    # 判断目录是否存在，不存在则创建
    if not os.path.isdir(logging_file):
        os.mkdir(logging_file)
    
    # 自定义一个目录和日志文件
    flask_file_handler = logging.handlers.RotatingFileHandler(filename=os.path.join(logging_file,'shopping.log'),maxBytes=max_logging_size,
                                         backupCount=logging_file_backup,encoding='utf-8')
  
    # 给当前的handler设置格式
    flask_file_handler.setFormatter(logging_format)
    
    # 得到一个logger对象
    logger = logging.getLogger('shopping')
    logger.addHandler(flask_file_handler)
    logger.setLevel(LOGGING_LEVEL)
    
    # 整个项目需要两个handler：文件、控制台
    flask_console_handler = logging.StreamHandler()
    flask_console_handler.setFormatter(logging.Formatter('[%(asctime)s] %(levelname)s %(module)s %(lineno)d : %(message)s'))
    
    # 当项目运行环境是debug，才用控制台输出
    if app.debug:
        logger.addHandler(flask_console_handler)